package com.zzx.system.service.impl;

import cn.hutool.core.util.IdUtil;
import cn.hutool.json.JSONUtil;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.zzx.common.annotation.DataScope;
import com.zzx.common.constant.UserConstants;
import com.zzx.common.core.domain.entity.SysDept;
import com.zzx.common.core.domain.entity.SysRole;
import com.zzx.common.core.domain.entity.SysUser;
import com.zzx.common.exception.ServiceException;
import com.zzx.config.UserContext;
import com.zzx.entity.TCompany;
import com.zzx.entity.vo.DecryptUserInfo;
import com.zzx.entity.vo.WxLoginResult;
import com.zzx.service.TCompanyService;
import com.zzx.utils.HttpClientUtil;
import com.zzx.utils.JWTUtil;
import com.zzx.utils.StringUtils;
import com.zzx.utils.QRCodeGenerator;
import com.zzx.utils.bean.BeanValidators;
import com.zzx.utils.spring.SpringUtils;
import com.zzx.system.domain.SysPost;
import com.zzx.system.domain.SysUserPost;
import com.zzx.system.domain.SysUserRole;
import com.zzx.system.mapper.*;
import com.zzx.system.service.ISysDeptService;
import com.zzx.system.service.ISysUserService;
import org.apache.logging.log4j.util.Strings;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;
import org.springframework.util.ObjectUtils;

import javax.annotation.Resource;
import javax.crypto.Cipher;
import javax.crypto.spec.IvParameterSpec;
import javax.crypto.spec.SecretKeySpec;
import javax.validation.Validator;
import java.nio.charset.StandardCharsets;
import java.util.*;
import java.util.stream.Collectors;

/**
 * 用户 业务层处理
 *
 * @author ruoyi
 */
@Service
public class SysUserServiceImpl extends ServiceImpl<SysUserMapper, SysUser> implements ISysUserService {
    private static final Logger log = LoggerFactory.getLogger(SysUserServiceImpl.class);

    @Autowired
    private SysUserMapper userMapper;

    @Autowired
    private SysRoleMapper roleMapper;

    @Autowired
    private SysPostMapper postMapper;

    @Autowired
    private SysUserRoleMapper userRoleMapper;

    @Autowired
    private SysUserPostMapper userPostMapper;

//    @Autowired
//    private ISysConfigService configService;

    @Autowired
    private ISysDeptService deptService;

    @Autowired
    protected Validator validator;



    @Value("${wechat.appId}")
    private String appId;

    @Value("${wechat.secret}")
    private String secret;

    @Value("${wechat.grantType}")
    private String grantType;

    @Resource
    private TCompanyService tCompanyService;

    /**
     * 根据条件分页查询用户列表
     *
     * @param user 用户信息
     * @return 用户信息集合信息
     */
    @Override
    @DataScope(deptAlias = "d", userAlias = "u")
    public List<SysUser> selectUserList(SysUser user) {
        user.setCompanyId(UserContext.getCompanyId());
        return userMapper.selectUserList(user);
    }

    /**
     * 根据条件分页查询已分配用户角色列表
     *
     * @param user 用户信息
     * @return 用户信息集合信息
     */
    @Override
    @DataScope(deptAlias = "d", userAlias = "u")
    public List<SysUser> selectAllocatedList(SysUser user) {
        return userMapper.selectAllocatedList(user);
    }

    /**
     * 根据条件分页查询未分配用户角色列表
     *
     * @param user 用户信息
     * @return 用户信息集合信息
     */
    @Override
    @DataScope(deptAlias = "d", userAlias = "u")
    public List<SysUser> selectUnallocatedList(SysUser user) {
        return userMapper.selectUnallocatedList(user);
    }

    /**
     * 通过用户名查询用户
     *
     * @param userName 用户名
     * @return 用户对象信息
     */
    @Override
    public SysUser selectUserByUserName(String userName) {
        return userMapper.selectUserByUserName(userName);
    }

    /**
     * 通过用户ID查询用户
     *
     * @param userId 用户ID
     * @return 用户对象信息
     */
    @Override
    public SysUser selectUserById(Long userId) {
        return userMapper.selectUserById(userId);
    }

    /**
     * 查询用户所属角色组
     *
     * @param userName 用户名
     * @return 结果
     */
    @Override
    public String selectUserRoleGroup(String userName) {
        List<SysRole> list = roleMapper.selectRolesByUserName(userName);
        if (CollectionUtils.isEmpty(list)) {
            return StringUtils.EMPTY;
        }
        return list.stream().map(SysRole::getRoleName).collect(Collectors.joining(","));
    }

    /**
     * 查询用户所属岗位组
     *
     * @param userName 用户名
     * @return 结果
     */
    @Override
    public String selectUserPostGroup(String userName) {
        List<SysPost> list = postMapper.selectPostsByUserName(userName);
        if (CollectionUtils.isEmpty(list)) {
            return StringUtils.EMPTY;
        }
        return list.stream().map(SysPost::getPostName).collect(Collectors.joining(","));
    }

    /**
     * 校验用户名称是否唯一
     *
     * @param user 用户信息
     * @return 结果
     */
    @Override
    public boolean checkUserNameUnique(SysUser user) {
        Long userId = StringUtils.isNull(user.getUserId()) ? -1L : user.getUserId();
        SysUser info = userMapper.checkUserNameUnique(user.getUserName());
        if (StringUtils.isNotNull(info) && info.getUserId().longValue() != userId.longValue()) {
            return UserConstants.NOT_UNIQUE;
        }
        return UserConstants.UNIQUE;
    }

    /**
     * 校验手机号码是否唯一
     *
     * @param user 用户信息
     * @return
     */
    @Override
    public boolean checkPhoneUnique(SysUser user) {
        Long userId = StringUtils.isNull(user.getUserId()) ? -1L : user.getUserId();
        SysUser info = userMapper.checkPhoneUnique(user.getPhonenumber());
        if (StringUtils.isNotNull(info) && info.getUserId().longValue() != userId.longValue()) {
            return UserConstants.NOT_UNIQUE;
        }
        return UserConstants.UNIQUE;
    }

    /**
     * 校验email是否唯一
     *
     * @param user 用户信息
     * @return
     */
    @Override
    public boolean checkEmailUnique(SysUser user) {
        Long userId = StringUtils.isNull(user.getUserId()) ? -1L : user.getUserId();
        SysUser info = userMapper.checkEmailUnique(user.getEmail());
        if (StringUtils.isNotNull(info) && info.getUserId().longValue() != userId.longValue()) {
            return UserConstants.NOT_UNIQUE;
        }
        return UserConstants.UNIQUE;
    }

    /**
     * 校验用户是否允许操作
     *
     * @param user 用户信息
     */
    @Override
    public void checkUserAllowed(SysUser user) {
        if (StringUtils.isNotNull(user.getUserId()) && user.isAdmin()) {
            throw new ServiceException("不允许操作超级管理员用户");
        }
    }

    /**
     * 校验用户是否有数据权限
     *
     * @param userId 用户id
     */
    @Override
    public void checkUserDataScope(Long userId) {
//        if (!SysUser.isAdmin(SecurityUtils.getUserId()))//todo
        if (true)//todo
        {
            SysUser user = new SysUser();
            user.setUserId(userId);
            List<SysUser> users = SpringUtils.getAopProxy(this).selectUserList(user);
            if (StringUtils.isEmpty(users)) {
                throw new ServiceException("没有权限访问用户数据！");
            }
        }
    }

    /**
     * 新增保存用户信息
     *
     * @param user 用户信息
     * @return 结果
     */
    @Override
    @Transactional
    public int insertUser(SysUser user) {
        // 新增用户信息
        user.setCompanyId(UserContext.getCompanyId());
        int rows = userMapper.insertUser(user);
        // 新增用户岗位关联
        insertUserPost(user);
        // 新增用户与角色管理
        insertUserRole(user);
        return rows;
    }

    /**
     * 注册用户信息
     *
     * @param user 用户信息
     * @return 结果
     */
    @Override
    public boolean registerUser(SysUser user) {
        user.setCompanyId(UserContext.getCompanyId());
        return userMapper.insertUser(user) > 0;
    }

    /**
     * 修改保存用户信息
     *
     * @param user 用户信息
     * @return 结果
     */
    @Override
    @Transactional
    public int updateUser(SysUser user) {
        Long userId = user.getUserId();
        // 删除用户与角色关联
        userRoleMapper.deleteUserRoleByUserId(userId);
        // 新增用户与角色管理
        insertUserRole(user);
        // 删除用户与岗位关联
        userPostMapper.deleteUserPostByUserId(userId);
        // 新增用户与岗位管理
        insertUserPost(user);
        return userMapper.updateUser(user);
    }

    /**
     * 用户授权角色
     *
     * @param userId  用户ID
     * @param roleIds 角色组
     */
    @Override
    @Transactional
    public void insertUserAuth(Long userId, Long[] roleIds) {
        userRoleMapper.deleteUserRoleByUserId(userId);
        insertUserRole(userId, roleIds);
    }

    /**
     * 修改用户状态
     *
     * @param user 用户信息
     * @return 结果
     */
    @Override
    public int updateUserStatus(SysUser user) {
        return userMapper.updateUser(user);
    }

    /**
     * 修改用户基本信息
     *
     * @param user 用户信息
     * @return 结果
     */
    @Override
    public int updateUserProfile(SysUser user) {
        return userMapper.updateUser(user);
    }

    /**
     * 修改用户头像
     *
     * @param userId 用户ID
     * @param avatar 头像地址
     * @return 结果
     */
    @Override
    public boolean updateUserAvatar(Long userId, String avatar) {
        return userMapper.updateUserAvatar(userId, avatar) > 0;
    }

    /**
     * 重置用户密码
     *
     * @param user 用户信息
     * @return 结果
     */
    @Override
    public int resetPwd(SysUser user) {
        return userMapper.updateUser(user);
    }

    /**
     * 重置用户密码
     *
     * @param userId   用户ID
     * @param password 密码
     * @return 结果
     */
    @Override
    public int resetUserPwd(Long userId, String password) {
        return userMapper.resetUserPwd(userId, password);
    }

    /**
     * 新增用户角色信息
     *
     * @param user 用户对象
     */
    public void insertUserRole(SysUser user) {
        this.insertUserRole(user.getUserId(), user.getRoleIds());
    }

    /**
     * 新增用户岗位信息
     *
     * @param user 用户对象
     */
    public void insertUserPost(SysUser user) {
        Long[] posts = user.getPostIds();
        if (StringUtils.isNotEmpty(posts)) {
            // 新增用户与岗位管理
            List<SysUserPost> list = new ArrayList<SysUserPost>(posts.length);
            for (Long postId : posts) {
                SysUserPost up = new SysUserPost();
                up.setUserId(user.getUserId());
                up.setPostId(postId);
                list.add(up);
            }
            userPostMapper.batchUserPost(list);
        }
    }

    /**
     * 新增用户角色信息
     *
     * @param userId  用户ID
     * @param roleIds 角色组
     */
    public void insertUserRole(Long userId, Long[] roleIds) {
        if (StringUtils.isNotEmpty(roleIds)) {
            // 新增用户与角色管理
            List<SysUserRole> list = new ArrayList<SysUserRole>(roleIds.length);
            for (Long roleId : roleIds) {
                SysUserRole ur = new SysUserRole();
                ur.setUserId(userId);
                ur.setRoleId(roleId);
                list.add(ur);
            }
            userRoleMapper.batchUserRole(list);
        }
    }

    /**
     * 通过用户ID删除用户
     *
     * @param userId 用户ID
     * @return 结果
     */
    @Override
    @Transactional
    public int deleteUserById(Long userId) {
        // 删除用户与角色关联
        userRoleMapper.deleteUserRoleByUserId(userId);
        // 删除用户与岗位表
        userPostMapper.deleteUserPostByUserId(userId);
        return userMapper.deleteUserById(userId);
    }

    /**
     * 批量删除用户信息
     *
     * @param userIds 需要删除的用户ID
     * @return 结果
     */
    @Override
    @Transactional
    public int deleteUserByIds(Long[] userIds) {
        for (Long userId : userIds) {
            checkUserAllowed(new SysUser(userId));
            checkUserDataScope(userId);
        }
        // 删除用户与角色关联
        userRoleMapper.deleteUserRole(userIds);
        // 删除用户与岗位关联
        userPostMapper.deleteUserPost(userIds);
        return userMapper.deleteUserByIds(userIds);
    }

    /**
     * 导入用户数据
     *
     * @param userList        用户数据列表
     * @param isUpdateSupport 是否更新支持，如果已存在，则进行更新数据
     * @param operName        操作用户
     * @return 结果
     */
    @Override
    public String importUser(List<SysUser> userList, Boolean isUpdateSupport, String operName) {
        if (StringUtils.isNull(userList) || userList.size() == 0) {
            throw new ServiceException("导入用户数据不能为空！");
        }
        int successNum = 0;
        int failureNum = 0;
        StringBuilder successMsg = new StringBuilder();
        StringBuilder failureMsg = new StringBuilder();
        for (SysUser user : userList) {
            try {
                // 验证是否存在这个用户
                SysUser u = userMapper.selectUserByUserName(user.getUserName());
                if (StringUtils.isNull(u)) {
                    BeanValidators.validateWithException(validator, user);
                    deptService.checkDeptDataScope(user.getDeptId());
//                    String password = configService.selectConfigByKey("sys.user.initPassword");
//                    user.setPassword(SecurityUtils.encryptPassword(password));//todo
                    user.setCreateBy(operName);
                    user.setCompanyId(UserContext.getCompanyId());
                    userMapper.insertUser(user);
                    successNum++;
                    successMsg.append("<br/>" + successNum + "、账号 " + user.getUserName() + " 导入成功");
                } else if (isUpdateSupport) {
                    BeanValidators.validateWithException(validator, user);
                    checkUserAllowed(u);
                    checkUserDataScope(u.getUserId());
                    deptService.checkDeptDataScope(user.getDeptId());
                    user.setUserId(u.getUserId());
                    user.setUpdateBy(operName);
                    userMapper.updateUser(user);
                    successNum++;
                    successMsg.append("<br/>" + successNum + "、账号 " + user.getUserName() + " 更新成功");
                } else {
                    failureNum++;
                    failureMsg.append("<br/>" + failureNum + "、账号 " + user.getUserName() + " 已存在");
                }
            } catch (Exception e) {
                failureNum++;
                String msg = "<br/>" + failureNum + "、账号 " + user.getUserName() + " 导入失败：";
                failureMsg.append(msg + e.getMessage());
                log.error(msg, e);
            }
        }
        if (failureNum > 0) {
            failureMsg.insert(0, "很抱歉，导入失败！共 " + failureNum + " 条数据格式不正确，错误如下：");
            throw new ServiceException(failureMsg.toString());
        } else {
            successMsg.insert(0, "恭喜您，数据已全部导入成功！共 " + successNum + " 条，数据如下：");
        }
        return successMsg.toString();
    }




    @Override
    public SysUser getByOpenId(String openId) {
        return this.getOne(new LambdaQueryWrapper<SysUser>().eq(SysUser::getOpenId, openId));
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public JSONObject wxLogin(String code) {
        if (Strings.isBlank(code)) {
            throw new RuntimeException("请传code");
        }
//        // 1. 调用微信接口获取session_key和openid
        String url = "https://api.weixin.qq.com/sns/jscode2session";
        Map<String, String> params = new HashMap<>();
        params.put("appid", appId);
        params.put("secret", secret);
        params.put("js_code", code);
        params.put("grant_type", grantType);
        String result = HttpClientUtil.sendGet(url, params);
//        String result = "{\"session_key\":\"fTng+tmvxwYVHqRRvlFDJw==\",\"openid\":\"oW0Pw605ISTCR30YUOj6-XtsmDQY\"}";
        WxLoginResult wxLoginResult = JSONUtil.toBean(result, WxLoginResult.class);

        if (!ObjectUtils.isEmpty((wxLoginResult.getErrcode()))) {
            throw new RuntimeException("Invalid code");
        }

        // 2. 根据openid查询用户是否存在
        SysUser user = this.getByOpenId(wxLoginResult.getOpenid());

        // 3. 如果用户不存在则创建新用户
        if (user == null) {
            user = new SysUser();
            user.setUserCategory(1);
            user.setOpenId(wxLoginResult.getOpenid());
            user.setUserName("计件用户");
//            user.setRole(RoleEnum.EMPLOYEE.getCode());
//            user.setSessionKey(wxLoginResult.getSession_key());
//            user.setUnionId(wxLoginResult.getUnionid());
            user.setCreateTime(new Date());
            user.setUpdateTime(new Date());
            user.setUserId(IdUtil.getSnowflakeNextId());
            this.save(user);
        } else {
            // 更新用户信息
//            user.setUpdateTime(LocalDateTime.now());
//            user.setOpenId(wxLoginResult.getOpenid());
//            user.setSessionKey(wxLoginResult.getSession_key());
//            this.updateById(user);
        }

        // 4. 生成自定义登录态token
        Map<String, Object> claims = new HashMap<>();
        claims.put("openId", user.getOpenId());
        claims.put("userId", user.getUserId());
        claims.put("userCategory", user.getUserCategory());
        if (!ObjectUtils.isEmpty(user.getCompanyId())) {
            claims.put("companyId", user.getCompanyId());
        }
        String token = JWTUtil.createToken(claims);
        log.info("openId>>{},token>>{}",user.getOpenId(),token);
        JSONObject res = new JSONObject();
        res.put("token",token);
        res.put("openId",user.getOpenId());
        res.put("userInfo",user);
        return res;
    }

    @Transactional(rollbackFor = Exception.class)
    @Override
    public void add(SysUser param) {
        LambdaQueryWrapper<SysUser> wrapper = new LambdaQueryWrapper<>(SysUser.class);
        wrapper.eq(SysUser::getCompanyId, UserContext.getCompanyId());
        wrapper.eq(SysUser::getUserName, param.getUserName());
        if (ObjectUtils.isEmpty(param.getUserId())) {
            param.setUserId(IdUtil.getSnowflakeNextId());
            param.setCreateBy(String.valueOf(UserContext.getUserId()));
            param.setCreateTime(new Date());
        } else {
            wrapper.ne(SysUser::getUserId, param.getUserId());
            param.setUpdateBy(String.valueOf(UserContext.getUserId()));
            param.setUpdateTime(new Date());
        }
        List<SysUser> tUsers = this.getBaseMapper().selectList(wrapper);
        if (!tUsers.isEmpty()) {
            throw new RuntimeException("名称重复");
        }
        this.saveOrUpdate(param);
    }

    @Override
    public List<SysUser> selectBy(SysUser param) {

        LambdaQueryWrapper<SysUser> wrapper = new LambdaQueryWrapper<>(SysUser.class);
        if (!ObjectUtils.isEmpty(param.getUserName())) {
            wrapper.like(SysUser::getUserName, param.getUserName());
        }
        if (!ObjectUtils.isEmpty(param.getUserNum())) {
            wrapper.like(SysUser::getUserNum, param.getUserNum());
        }
        if (!ObjectUtils.isEmpty(param.getUserZw())) {
            wrapper.eq(SysUser::getUserZw, param.getUserZw());
        }
        if (!ObjectUtils.isEmpty(param.getUserId())) {
            wrapper.eq(SysUser::getUserId, param.getUserId());
        }
        wrapper.eq(SysUser::getCompanyId, UserContext.getCompanyId());
        List<SysUser> tTUserList = this.getBaseMapper().selectList(wrapper);

//        MPJLambdaWrapper<TUser> mpjLambdaWrapper = new MPJLambdaWrapper();
//        mpjLambdaWrapper.selectAll(TUser.class);
//        mpjLambdaWrapper.select(TCompany::getCompanyName);
//        mpjLambdaWrapper.leftJoin(TCompany.class, TCompany::getCompanyId, TUser::getCompanyId);
//        this.getBaseMapper().selectJoinList(TUser.class, mpjLambdaWrapper);
        return tTUserList;
    }

    @Override
    @Transactional (rollbackFor = Exception.class)
    public void deleteById(SysUser param) {
        this.removeById(param.getUserId());
    }

    @Override
    public SysUser findById(Long userId) {
        SysUser sysUser = this.getById(userId);
        return sysUser;
    }

    @Override
    public JSONObject decryptUserInfo(DecryptUserInfo decryptUserInfo) {
//        JSONObject jsonObject = new JSONObject();
//        try {
//            byte[] key = Base64.getDecoder().decode(decryptUserInfo.getSessionKey());
//            byte[] ivBytes = Base64.getDecoder().decode(decryptUserInfo.getIv());
//            Cipher cipher = Cipher.getInstance("AES/CBC/PKCS7Padding");
//            cipher.init(Cipher.DECRYPT_MODE, new SecretKeySpec(key, "AES"), new IvParameterSpec(ivBytes));
//            byte[] result = cipher.doFinal(Base64.getDecoder().decode(decryptUserInfo.getEncryptedData()));
//            jsonObject = JSONObject.parseObject(new String(result, StandardCharsets.UTF_8));
//        } catch (NoSuchAlgorithmException e) {
//            e.printStackTrace();
//            throw new RuntimeException(e);
//        } catch (NoSuchPaddingException e) {
//            e.printStackTrace();
//            throw new RuntimeException(e);
//        } catch (InvalidKeyException e) {
//            e.printStackTrace();
//            throw new RuntimeException(e);
//        } catch (InvalidAlgorithmParameterException e) {
//            e.printStackTrace();
//            throw new RuntimeException(e);
//        } catch (IllegalBlockSizeException e) {
//            e.printStackTrace();
//            throw new RuntimeException(e);
//        } catch (BadPaddingException e) {
//            e.printStackTrace();
//            throw new RuntimeException(e);
//        }
//        return jsonObject;
        String openId = UserContext.getOpenId();
        String sessionKey = UserContext.getSessionKey();
        String s = decryptData(decryptUserInfo.getEncryptedData(), decryptUserInfo.getSessionKey(), decryptUserInfo.getIv());
        return JSONObject.parseObject(s);
    }

    private String cleanBase64String(String input) {
        if (input == null || input.isEmpty()) {
            return input;
        }

        // 移除非 Base64 合法字符
        String cleaned = input.replaceAll("[^A-Za-z0-9+/]", "");

        // 使用 StringBuilder 替代 repeat()
        int padding = 4 - (cleaned.length() % 4);
        if (padding < 4) {
            StringBuilder sb = new StringBuilder(cleaned);
            for (int i = 0; i < padding; i++) {
                sb.append('=');
            }
            cleaned = sb.toString();
        }

        return cleaned;
    }


    /**
     * 解密微信小程序加密数据
     */
    public String decryptData(String encryptedData, String sessionKey, String iv) {

        try {
            encryptedData = cleanBase64String(encryptedData);
            sessionKey = cleanBase64String(sessionKey);
            iv = cleanBase64String(iv);
            byte[] encryptedBytes = Base64.getDecoder().decode(encryptedData);
            byte[] keyBytes = Base64.getDecoder().decode(sessionKey);
            byte[] ivBytes = Base64.getDecoder().decode(iv);

            if (keyBytes.length != 16 || ivBytes.length != 16) {
                throw new IllegalArgumentException("密钥或初始向量长度不正确");
            }

            Cipher cipher = Cipher.getInstance("AES/CBC/PKCS5Padding");
            SecretKeySpec keySpec = new SecretKeySpec(keyBytes, "AES");
            IvParameterSpec ivSpec = new IvParameterSpec(ivBytes);

            cipher.init(Cipher.DECRYPT_MODE, keySpec, ivSpec);
            byte[] decryptedBytes = cipher.doFinal(encryptedBytes);

            return new String(decryptedBytes, StandardCharsets.UTF_8);
        } catch (Exception e) {
            e.printStackTrace();
            throw new RuntimeException("数据解密失败", e);
        }
    }

    // 解密后数据结构
//    {
//        "nickName": "微信用户",
//            "gender": 1,
//            "avatarUrl": "https://thirdwx.qlogo.cn/...",
//            "country": "中国",
//            "province": "北京",
//            "city": "朝阳区"
//    }

    @Resource
    private QRCodeGenerator qrCodeGenerator;
    @Override
    @Transactional(rollbackFor = Exception.class)
    public JSONObject editTUserAndTCompany(SysUser param) {
        TCompany tCompany = param.getTCompany();
        if (ObjectUtils.isEmpty(tCompany.getCompanyId())) {
            tCompany.setCompanyId(IdUtil.getSnowflakeNextId());
            byte[] bytes = qrCodeGenerator.generateAndSaveQRCode(tCompany.getCompanyId().toString());
            tCompany.setCompanyQrCode(bytes);

        }
        // 保存公司信息
        tCompanyService.saveOrUpdate(tCompany);

        // 保存用户信息
        SysUser sysUser = this.getBaseMapper().selectById(param.getUserId());
        sysUser.setCompanyId(tCompany.getCompanyId());
        sysUser.setUserName(param.getUserName());
        sysUser.setUserPhone(param.getUserPhone());
//        sysUser.setRole(RoleEnum.ADMIN.getCode());
        this.saveOrUpdate(sysUser);
        //预制顶级部门
        SysDept dept = new SysDept();
        dept.setDeptId(IdUtil.getSnowflakeNextId());
        dept.setParentId(0L);
        dept.setAncestors("0");
        dept.setDeptName(tCompany.getCompanyName());
        dept.setOrderNum(0);
        dept.setStatus("0");
        dept.setDelFlag("0");
        dept.setCreateBy(String.valueOf(sysUser.getUserId()));
        dept.setCreateTime(new Date());
        dept.setCompanyId(tCompany.getCompanyId());
        dept.setRoot( true);
        deptService.insertDept(dept);

        // 生成token
        Map<String, Object> claims = new HashMap<>();
        claims.put("openId", sysUser.getOpenId());
        claims.put("userId", sysUser.getUserId());
        claims.put("userCategory", sysUser.getUserCategory());
        claims.put("companyId", tCompany.getCompanyId());
        String token = JWTUtil.createToken(claims);
        JSONObject res = new JSONObject();
        res.put("token",token);
        res.put("openId",sysUser.getOpenId());
        res.put("userInfo",sysUser);
        return res;
    }
}
